package cool.goodbai.handler;

import cn.dev33.satoken.stp.StpUtil;
import cool.goodbai.api.BaseEntity;
import cool.goodbai.entity.User;
import org.springframework.stereotype.Component;

import java.util.Date;

@Component
public class DataProxy {
    public void preInsert(BaseEntity data) {
        data.setUtcCreate(new Date());
        data.setUtcModify(data.getUtcCreate());
        if (StpUtil.getSession().getModel("user", User.class) != null) {
            data.setCreateBy(StpUtil.getSession().getModel("user", User.class).getId().toString());
            data.setModifyBy(data.getCreateBy());
        }
    }

    public void preUpdate(BaseEntity data) {
        data.setUtcModify(new Date());
        data.setModifyBy(StpUtil.getSession().getModel("user", User.class).getId().toString());
    }
}
